# 1-导包
import requests
from config import base_url


# 2-定义类（模块）
class ApiCantract:

    # 3-定义初始化方法，定义接口测试静态数据--》url
    def __init__(self):
        # 上传合同url
        self.upload_url = base_url+"/api/common/upload"
        # 新增合同url
        self.add_contrat = base_url+"/api/contract"

    # 4-定义方法（一个方法一个接口）,data是一个二进制打开的文件对象
    def api_cantract_upload(self,data,token):
        # 定义请求参数
        req_data = {"file": data}
        # 定义请求头
        req_header = {"Authorization":token}
        # 执行接口请求，获取响应对象
        resp = requests.post(url=self.upload_url,files=req_data,headers=req_header)
        # 返回响应对象
        return resp

    # 5-合同新增
    def add_contract(self, test_data, token):
        return requests.post(url=self.add_contrat, json=test_data,
                             headers={"Authorization": token})
